from torchsummary import summary
from torchvision import models


def show_vgg_struction():
    vgg = models.vgg16()
    print(vgg)
    summary(vgg, (3, 224, 224), batch_size=2048)


if __name__ == '__main__':
    show_vgg_struction()
